home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 2001 December / pcwk12201b.iso / Wersje pelne i specjalne / Winamp 2.77 i 3.0beta / wasabi-sdk_beta1.exe / jnetlib / sercon.cpp < prev    next >
C/C++ Source or Header  |  2001-10-08  |  8KB  |  342 lines
  1. /*
  2.  
  3.   Nullsoft WASABI Source File License
  4.  
  5.   Copyright 1999-2001 Nullsoft, Inc.
  6.  
  7.     This software is provided 'as-is', without any express or implied
  8.     warranty.  In no event will the authors be held liable for any damages
  9.     arising from the use of this software.
  10.  
  11.     Permission is granted to anyone to use this software for any purpose,
  12.     including commercial applications, and to alter it and redistribute it
  13.     freely, subject to the following restrictions:
  14.  
  15.     1. The origin of this software must not be misrepresented; you must not
  16.        claim that you wrote the original software. If you use this software
  17.        in a product, an acknowledgment in the product documentation would be
  18.        appreciated but is not required.
  19.     2. Altered source versions must be plainly marked as such, and must not be
  20.        misrepresented as being the original software.
  21.     3. This notice may not be removed or altered from any source distribution.
  22.  
  23.  
  24.   Brennan Underwood
  25.   brennan@nullsoft.com
  26.  
  27. */
  28.  
  29. /*
  30. ** JNetLib
  31. ** Copyright (C) 2000-2001 Nullsoft, Inc.
  32. ** Author: Justin Frankel
  33. ** File: sercon.cpp
  34. ** License: see jnetlib.h
  35. */
  36.  
  37. #include "netinc.h"
  38. #include "util.h"
  39. #include "sercon.h"
  40.  
  41.  
  42. JNL_SerCon::JNL_SerCon(int sendbufsize, int recvbufsize)
  43. {
  44.   m_errorstr="";
  45.   m_recv_buffer_len=recvbufsize;
  46.   m_send_buffer_len=sendbufsize;
  47.   m_recv_buffer=(char*)malloc(m_recv_buffer_len);
  48.   m_send_buffer=(char*)malloc(m_send_buffer_len);
  49.   m_file=-1;
  50.   memset(m_recv_buffer,0,recvbufsize);
  51.   memset(m_send_buffer,0,sendbufsize);
  52.   m_recv_len=m_recv_pos=0;
  53.   m_send_len=m_send_pos=0;
  54.   m_err=1; // 1 = not connected
  55. }
  56.  
  57. void JNL_SerCon::connect(char *dev)
  58. {
  59.   close();
  60.   m_file = open(dev,O_RDWR);
  61.   if (m_file < 0)
  62.   {
  63.     m_err=2;
  64.     m_errorstr = "error opening device";
  65.   }
  66.   else 
  67.   {
  68.     SET_SOCK_BLOCK(m_file,0);
  69.     m_err=0;
  70.   }
  71. }
  72.  
  73. JNL_SerCon::~JNL_SerCon()
  74. {
  75.   if (m_file >= 0)
  76.   {
  77.     ::close(m_file);
  78.     m_file=-1;
  79.   }
  80.   free(m_recv_buffer);
  81.   free(m_send_buffer);
  82. }
  83.  
  84. void JNL_SerCon::run(int max_send_bytes, int max_recv_bytes, int *bytes_sent, int *bytes_rcvd)
  85. {
  86.   int bytes_allowed_to_send=(max_send_bytes<0)?m_send_buffer_len:max_send_bytes;
  87.   int bytes_allowed_to_recv=(max_recv_bytes<0)?m_recv_buffer_len:max_recv_bytes;
  88.  
  89.   if (bytes_sent) *bytes_sent=0;
  90.   if (bytes_rcvd) *bytes_rcvd=0;
  91.  
  92.   if (m_err) return;
  93.  
  94.   if (m_send_len>0 && bytes_allowed_to_send>0)
  95.   {
  96.     int len=m_send_buffer_len-m_send_pos;
  97.     if (len > m_send_len) len=m_send_len;
  98.     if (len > bytes_allowed_to_send) len=bytes_allowed_to_send;
  99.     if (len > 0)
  100.     {
  101.       int res=::write(m_file,m_send_buffer+m_send_pos,len);
  102.       if (res==-1 && ERRNO != EWOULDBLOCK)
  103.       {            
  104.         m_err=2;
  105.         m_errorstr = "write() returned error";
  106.         return;
  107.       }
  108.       if (res>0)
  109.       {
  110.         bytes_allowed_to_send-=res;
  111.         if (bytes_sent) *bytes_sent+=res;
  112.         m_send_pos+=res;
  113.         m_send_len-=res;
  114.       }
  115.     }
  116.     if (m_send_pos>=m_send_buffer_len) 
  117.     {
  118.       m_send_pos=0;
  119.       if (m_send_len>0)
  120.       {
  121.         len=m_send_buffer_len-m_send_pos;
  122.         if (len > m_send_len) len=m_send_len;
  123.         if (len > bytes_allowed_to_send) len=bytes_allowed_to_send;
  124.         int res=::write(m_file,m_send_buffer+m_send_pos,len);
  125.         if (res==-1 && ERRNO != EWOULDBLOCK)
  126.         {
  127.           m_err=2;
  128.           m_errorstr = "write() returned error";
  129.           return;
  130.         }
  131.         if (res>0)
  132.         {
  133.           bytes_allowed_to_send-=res;
  134.           if (bytes_sent) *bytes_sent+=res;
  135.           m_send_pos+=res;
  136.           m_send_len-=res;
  137.         }
  138.       }
  139.     }
  140.   }
  141.   if (m_recv_len<m_recv_buffer_len)
  142.   {
  143.     int len=m_recv_buffer_len-m_recv_pos;
  144.     if (len > m_recv_buffer_len-m_recv_len) len=m_recv_buffer_len-m_recv_len;
  145.     if (len > bytes_allowed_to_recv) len=bytes_allowed_to_recv;
  146.     if (len>0)
  147.     {
  148.       int res=::read(m_file,m_recv_buffer+m_recv_pos,len);
  149.       if (res == 0 || (res < 0 && ERRNO != EWOULDBLOCK))
  150.       {        
  151.         m_err=2;
  152.         m_errorstr = "read() returned error";
  153.         return;
  154.       }
  155.       if (res > 0)
  156.       {
  157.         bytes_allowed_to_recv-=res;
  158.         if (bytes_rcvd) *bytes_rcvd+=res;
  159.         m_recv_pos+=res;
  160.         m_recv_len+=res;
  161.       }
  162.     }
  163.     if (m_recv_pos >= m_recv_buffer_len)
  164.     {
  165.        m_recv_pos=0;
  166.        if (m_recv_len < m_recv_buffer_len)
  167.        {
  168.          len=m_recv_buffer_len-m_recv_len;
  169.          if (len > bytes_allowed_to_recv) len=bytes_allowed_to_recv;
  170.          if (len > 0)
  171.          {
  172.            int res=::read(m_file,m_recv_buffer+m_recv_pos,len);
  173.            if (res == 0 || (res < 0 && ERRNO != EWOULDBLOCK))
  174.            {        
  175.              m_err=2;
  176.              m_errorstr = "read() returned error";
  177.              return;
  178.            }
  179.            if (res > 0)
  180.            {
  181.              bytes_allowed_to_recv-=res;
  182.              if (bytes_rcvd) *bytes_rcvd+=res;
  183.              m_recv_pos+=res;
  184.              m_recv_len+=res;
  185.            }
  186.          }
  187.        }
  188.     }
  189.   }
  190. }
  191.  
  192. void JNL_SerCon::close()
  193. {
  194.   if (m_file >= 0)
  195.   {
  196.     ::close(m_file);
  197.   }
  198.   m_file=-1;
  199.   memset(m_recv_buffer,0,m_recv_buffer_len);
  200.   memset(m_send_buffer,0,m_send_buffer_len);
  201.   m_recv_len=m_recv_pos=0;
  202.   m_send_len=m_send_pos=0;
  203. }
  204.  
  205. int JNL_SerCon::send_bytes_in_queue(void)
  206. {
  207.   return m_send_len;
  208. }
  209.  
  210. int JNL_SerCon::send_bytes_available(void)
  211. {
  212.   return m_send_buffer_len-m_send_len;
  213. }
  214.  
  215. int JNL_SerCon::send(char *data, int length)
  216. {
  217.   if (length > send_bytes_available())
  218.   {
  219.     return -1;
  220.   }
  221.   
  222.   int write_pos=m_send_pos+m_send_len;
  223.   if (write_pos >= m_send_buffer_len) 
  224.   {
  225.     write_pos-=m_send_buffer_len;
  226.   }
  227.  
  228.   int len=m_send_buffer_len-write_pos;
  229.   if (len > length) 
  230.   {
  231.     len=length;
  232.   }
  233.  
  234.   memcpy(m_send_buffer+write_pos,data,len);
  235.   if (length > len)
  236.   {
  237.     memcpy(m_send_buffer,data+len,length-len);
  238.   }
  239.   m_send_len+=length;
  240.   return 0;
  241. }
  242.  
  243. int JNL_SerCon::send_string(char *line)
  244. {
  245.   return send(line,strlen(line));
  246. }
  247.  
  248. int JNL_SerCon::recv_bytes_available(void)
  249. {
  250.   return m_recv_len;
  251. }
  252.  
  253. int JNL_SerCon::peek_bytes(char *data, int maxlength)
  254. {
  255.   if (maxlength > m_recv_len)
  256.   {
  257.     maxlength=m_recv_len;
  258.   }
  259.   int read_pos=m_recv_pos-m_recv_len;
  260.   if (read_pos < 0) 
  261.   {
  262.     read_pos += m_recv_buffer_len;
  263.   }
  264.   int len=m_recv_buffer_len-read_pos;
  265.   if (len > maxlength)
  266.   {
  267.     len=maxlength;
  268.   }
  269.   memcpy(data,m_recv_buffer+read_pos,len);
  270.   if (len < maxlength)
  271.   {
  272.     memcpy(data+len,m_recv_buffer,maxlength-len);
  273.   }
  274.  
  275.   return maxlength;
  276. }
  277.  
  278. int JNL_SerCon::recv_bytes(char *data, int maxlength)
  279. {
  280.   
  281.   int ml=peek_bytes(data,maxlength);
  282.   m_recv_len-=ml;
  283.   return ml;
  284. }
  285.  
  286. int JNL_SerCon::getbfromrecv(int pos, int remove)
  287. {
  288.   int read_pos=m_recv_pos-m_recv_len + pos;
  289.   if (pos < 0 || pos > m_recv_len) return -1;
  290.   if (read_pos < 0) 
  291.   {
  292.     read_pos += m_recv_buffer_len;
  293.   }
  294.   if (read_pos >= m_recv_buffer_len)
  295.   {
  296.     read_pos-=m_recv_buffer_len;
  297.   }
  298.   if (remove) m_recv_len--;
  299.   return m_recv_buffer[read_pos];
  300. }
  301.  
  302. int JNL_SerCon::recv_lines_available(void)
  303. {
  304.   int l=recv_bytes_available();
  305.   int lcount=0;
  306.   int lastch=0;
  307.   int pos;
  308.   for (pos=0; pos < l; pos ++)
  309.   {
  310.     int t=getbfromrecv(pos,0);
  311.     if (t == -1) return lcount;
  312.     if ((t=='\r' || t=='\n') &&(
  313.          (lastch != '\r' && lastch != '\n') || lastch==t
  314.         )) lcount++;
  315.     lastch=t;
  316.   }
  317.   return lcount;
  318. }
  319.  
  320. int JNL_SerCon::recv_line(char *line, int maxlength)
  321. {
  322.   if (maxlength > m_recv_len) maxlength=m_recv_len;
  323.   while (maxlength--)
  324.   {
  325.     int t=getbfromrecv(0,1);
  326.     if (t == -1) 
  327.     {
  328.       *line=0;
  329.       return 0;
  330.     }
  331.     if (t == '\r' || t == '\n')
  332.     {
  333.       int r=getbfromrecv(0,0);
  334.       if ((r == '\r' || r == '\n') && r != t) getbfromrecv(0,1);
  335.       *line=0;
  336.       return 0;
  337.     }
  338.     *line++=(char)t;
  339.   }
  340.   return 1;
  341. }
  342.